#pragma once

#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/time.h>
#include <atomic>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp/";

    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            // 获取时间
            gettimeofday(&_time, nullptr);
            // 获取秒数...
            return std::to_string(_time.tv_sec);
        }

        // 获取毫秒级别
        static std::string GetMilliSecond()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            /*                         秒                        微秒           */
            return std::to_string((_time.tv_sec * 1000) + (_time.tv_usec / 1000));
        }
    };

    // 和路径有关的类
    class PathUtil
    {
    public:
        // 添加后缀
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string res = temp_path;
            res += file_name;
            res += suffix;
            return res;
        }
        // 返回带路径的源文件 .cpp .c等
        //./temp/1234.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 返回带路径的可执行文件 .exe等
        static std::string Exc(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        // 返回带路径的错误存储文件 .strerr
        // 编译时错误
        static std::string CompileStderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_stderr");
        }

        // 运行时要用的文件
        // 输入流
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        // 输出流
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }

        // 运行时错误
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        static bool isFilexist(const std::string &path_name)
        {
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
            {
                // 找到了
                return true;
            }
            return false;
        }

        static std::string UniqueName()
        {
            // 要获取唯一的代码就要做到唯一标识
            // 我们可以使用时间戳 + 原子性
            static std::atomic<int> id(0);
            std::string time = TimeUtil::GetMilliSecond();
            std::string uniq_id = std::to_string(id);
            return uniq_id + "_" + time;
        }

        static bool WriteFile(const std::string &target, std::string content)
        {
            // 打开流
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            // 写入文件
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        /***************************************
         * 输入：
         * target:打开的目标文件
         * keep:是否要保存\n
         * 输出：
         * bool:返回是否读取成功
         * content:返回读取后的上下文
         ****************************************/
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear();

            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // getline不会保存换行符\n
            // getline在这里进行了强制类型转换（内部自己定义的
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        /**************************
         *
         **************************/
        static void SplitString(const std::string &str, std::vector<std::string> *out, const std::string &sep)
        {
            //      输出型参数  分割的字符串     分隔符    是否压缩分割符（多打连续几个空格也可以压缩成一个）
            boost::split((*out), str, boost::is_any_of(sep), boost::token_compress_on);
        }
    };
}